from api.ihrm_login_api import IhrmLoginApi


# 发送 登录成功请求，获取令牌，拼接成 请求头 返回。
def get_header():
    # 测试数据
    req_body = {"mobile": "13800000002", "password": "123456"}

    # 调用 自己封装的 login方法，获取响应结果
    resp = IhrmLoginApi.login(req_body)

    token = resp.json().get("data")

    # 将 令牌 组织到 请求头中
    req_header = {"Authorization": token}

    # 必须返回
    return req_header


if __name__ == '__main__':
    get_header()
